package si.banst.tecajnice.web;

import java.util.List;

import javax.faces.model.ListDataModel;

import org.primefaces.model.SelectableDataModel;

import si.banst.tecajnice.entitete.Tecaj;

public class TecajDataModel extends ListDataModel<Tecaj> implements
		SelectableDataModel<Tecaj> {
	
	public TecajDataModel(List<Tecaj> tecaji) {
		super(tecaji);
	}

	@Override
	public Tecaj getRowData(String arg0) {
		@SuppressWarnings("unchecked")
		List<Tecaj> tecaji = (List<Tecaj>) getWrappedData();  
        
        for(Tecaj t : tecaji) {  
            if(t.getId().toString().equalsIgnoreCase(arg0))  
                return t;  
        }  
          
        return null;  
	}

	@Override
	public Object getRowKey(Tecaj tecaj) {
		return tecaj.getId().toString();
	}

}
